from appium.options.android import UiAutomator2Options
from dotenv import load_dotenv
import os

# 设置 Desired Capabilities
options = UiAutomator2Options()
options.platform_name = 'Android'
options.device_name = 'Android'  # 使用 MuMu 模拟器时，可以设为 'Android'
options.udid = '127.0.0.1:7555'  # MuMu 模拟器的默认端口
options.no_reset = False
options.set_capability('adbExecTimeout', 60000)  # 添加超时配置
options.set_capability('uiautomator2ServerInstallTimeout', 60000)
options.set_capability('uiautomator2ServerLaunchTimeout', 60000)

# 加载环境变量
load_dotenv()  # ← 读取 .env，设置 os.environ
ENV = os.getenv("ENV", "development")  # ← 读取 ENV，没设置就默认 development

# 设置 driver 类型
DRIVER_TYPE = "web"  # 👈 关键配置！可选 "web" 或 "app"